package com.example.utils;

import com.example.annotation.Component;

import java.sql.Connection;
import java.sql.SQLException;

/**
 * 标记@Component：向集合中以类型名首字母小写注册
 * 普通类向集合中注册，不去校验该类是否进行接口实现
 *
 * @author kangshuai
 */
@Component
public class ConnectionUtils {

    /**
     * 存储当前线程的连接
     */
    private ThreadLocal<Connection> threadLocal = new ThreadLocal<>();


    /**
     * 从当前线程获取连接
     * 1、判断当前线程中是否已经绑定连接，如果没有绑定，需要从连接池获取一个连接绑定到当前线程
     * 2、从连接池拿连接
     * 3、绑定到当前线程
     *
     * @return
     * @throws SQLException
     */
    public Connection getCurrentThreadConn() throws SQLException {
        Connection connection = threadLocal.get();
        if (connection == null) {
            connection = DruidUtils.getInstance().getConnection();
            threadLocal.set(connection);
        }
        return connection;

    }
}